import cv2 as cv


def main():
    img = cv.imread('../data/test.png')
    if img is None:
        print("牛爷爷, 俺滴图图呢")
        exit(0)

    px = img[100, 100]
    print('1.1 像素坐标(100,100) 3个通道(B,G,R)的值:', px)

    blue = img[100, 100, 0]
    print('1.2 像素坐标(100,100) 第1个通道(B)的值:', blue)

    print('2.1 使用numpy优化函数访问像素坐标(100,100) 3个通道(B,G,R)的值:', img.item(100, 100, 0),
          img.item(100, 100, 1), img.item(100, 100, 2))

    print('2.2 使用数组索引设置像素坐标(100,100) 第3个通道(R)的值')
    img[100, 100, 2] = 100

    print('2.3 使用numpy优化函数访问像素坐标(100,100) 第3个通道(R)的值:', img.item(100, 100, 2))

    cv.imshow("Press Q to exit", img)
    while True:
        if cv.waitKey(0) == ord('q'):
            break
    cv.destroyAllWindows()


if __name__ == "__main__":
    main()
